from dotenv import load_dotenv
import os

load_dotenv()

MOONSHOT_CONFIG = {
    "base_url": os.getenv("MOONSHOT_URL"),
    "api_key": os.getenv("MOONSHOT_API_KEY"),
    "model_name": "moonshot-v1-8k",
    "temperature": 0.3
}

DOUBAO_CONFIG = {
    "base_url": "https://maas-api.ml-platform-cn-beijing.volces.com/v1/",
    "api_key": os.getenv("HUOSHAN_API_KEY"),
    "model_name": "Doubao-lite-4k",
}

if __name__ == "__main__":
    print(MOONSHOT_CONFIG)
    print(DOUBAO_CONFIG)